#!/bin/bash
CURRENT_DIR="$( cd "$( dirname "$0" )" && pwd )"
MEGATRON_PATCH_PATH=$( dirname $( dirname ${CURRENT_DIR}))
export PYTHONPATH=${MEGATRON_PATCH_PATH}:${MEGATRON_PATCH_PATH}/backends/megatron/Megatron-LM-250328:$PYTHONPATH

python ./convert_ckpt_qwen25_0.5B_hf2mg.py \
    --target-tensor-parallel-size 1 \
    --target-pipeline-parallel-size 1 \
    --load-dir /home/ltaishuang/PAI-megatron/qwen-ckpts/Qwen2.5-0.5B \
    --save-dir /home/ltaishuang/PAI-megatron/qwen-ckpts/Qwen2.5-0.5B.meg.pp1.tp1 \
    --num-layers 24
